Sorting algorithms visualized in python